/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package uk.ac.cam.can.watermarking;

import uk.ac.cam.can.watermarking.algorithm.AKH;
import uk.ac.cam.can.watermarking.algorithm.WatermarkingAlgorithm;
import uk.ac.cam.can.watermarking.misc.DataSet;

/**
 *
 * @author Thomas
 */
public class Watermark{
    public static final int AKH_ALGORITHM=0;
    
    private WatermarkingAlgorithm algo;
    private DataSet data;
    
    public Watermark(DataSet data, int type) throws Exception{
        this.data=data;
        switch(type){
            case AKH_ALGORITHM:
                algo= new AKH();
                break;
            default:
                throw new Exception("Algorithm is not set");
        }
    }
    
    public DataSet mark(long key){
        return algo.mark(data, key);
    }
    
    public float checkMark(long key){
        return algo.checkMark(data, key);
    }
}
